package com.gio.tools.common.config.datasource;

import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.springframework.stereotype.Component;

@Aspect
@Component
public class DataSourceAop {
    //在primary方法前执行
    @Before("execution(* com.gio.tools.modules.cdp..*.*(..))")
    public void setHive() {
        DataSourceType.setDataBaseType(DataSourceType.DataBaseType.HIVE);
    }

    //在secondary方法前执行
    @Before("execution(* com.gio.tools.modules.dsp..*.*(..))")
    public void setPg() {
        DataSourceType.setDataBaseType(DataSourceType.DataBaseType.PG);
    }
}